﻿// Copyright (c) Appeon. All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. 
// You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0.

using System;
using System.Data;
using System.Data.Common;
using JetBrains.Annotations;

namespace SnapObjects.Data.PostgreSql
{
    internal class PostgreSqlByteTypeMapping : ByteTypeMapping
    {
        public PostgreSqlByteTypeMapping(
            [NotNull] string dbTypeName, [CanBeNull] DbType? dbType = null)
            : base(dbTypeName, dbType)
        {
        }

        public override DbParameter CreateParameter(
            [NotNull] DbCommand command, 
            [NotNull] IAdoDbParameter adoDbParameter)
        {
            var dbParameter = base.CreateParameter(command, adoDbParameter);
            
            if (adoDbParameter.Value != null &&
                adoDbParameter.Value is Enum &&
                adoDbParameter.Direction != ParameterDirection.ReturnValue)
            {
                dbParameter.Value = Convert.ToByte(adoDbParameter.Value);
            }
            
            return dbParameter;
        }
    }
}
